package com.donghu.opanai;

import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.profile.ClientProfile;
import com.tencentcloudapi.common.profile.HttpProfile;
import com.tencentcloudapi.common.exception.TencentCloudSDKException;
import com.tencentcloudapi.nlp.v20190408.NlpClient;
import com.tencentcloudapi.nlp.v20190408.models.*;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.URL;

public class ChatBot
{
    public static void main(String [] args) {
//        try{
//            // 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
//            // 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
//            // 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
//            Credential cred = new Credential("AKIDoPl2GH8iolTv87sP7fx0uc7ORJo9JUXp", "8veGgvz2uYikYCk8wYw9B1f8RWP7dw11");
//            // 实例化一个http选项，可选的，没有特殊需求可以跳过
//            HttpProfile httpProfile = new HttpProfile();
//            httpProfile.setEndpoint("nlp.tencentcloudapi.com");
//            // 实例化一个client选项，可选的，没有特殊需求可以跳过
//            ClientProfile clientProfile = new ClientProfile();
//            clientProfile.setHttpProfile(httpProfile);
//            // 实例化要请求产品的client对象,clientProfile是可选的
//            NlpClient client = new NlpClient(cred, "ap-guangzhou", clientProfile);
//            // 实例化一个请求对象,每个接口都会对应一个request对象  1316063356
//            ChatBotRequest req = new ChatBotRequest();
//            req.setOpenId("1316063356");
//            req.setFlag(0L);
//            req.setQuery("");
//            // 返回的resp是一个ChatBotResponse的实例，与请求对象对应
//            ChatBotResponse resp = client.ChatBot(req);
//            // 输出json格式的字符串回包
//            System.out.println(ChatBotResponse.toJsonString(resp));
//        } catch (TencentCloudSDKException e) {
//            System.out.println(e.toString());
//        }
        String info = getQuestionInfo("java冒泡排序");
        System.out.println(info);
    }
    private static String getQuestionInfo(String question) {
        try {
            URL url = new URL("https://www.1bit.asia/openai/api/ask");
            HttpURLConnection con = (HttpURLConnection) url.openConnection();
            con.setRequestMethod("POST");
            // 设置请求头
            con.setRequestProperty("Content-Type", "application/json");
            // 设置请求体
            String requestBody = "{\"prompt\":\""+ question +"\",\"token\":\"apiuser005 3c6e8535-45e9-4000-ae86-ed03d6ed9600\",\"userName\":\"apiuser005\"}";
            con.setDoOutput(true);
            con.setDoInput(true);
            OutputStream os = con.getOutputStream();
            os.write(requestBody.getBytes());
            os.flush();
            os.close();
            BufferedReader in = new BufferedReader(new InputStreamReader(con.getInputStream()));
            String inputLine;
            StringBuffer content = new StringBuffer();
            while ((inputLine = in.readLine()) != null) {
                content.append(inputLine);
            }
            in.close();
            String questionInfo = content.toString();
            return questionInfo;
        } catch (IOException e) {
            e.printStackTrace();
            return "请求异常";
        }
    }
}